home
***
CD-ROM
|
disk
|
FTP
|
other
***
search
/
Amiga Tools 2
/
Amiga Tools 2.iso
/
tools
/
vim
/
src
/
memline.c
< prev
next >
Wrap
C/C++ Source or Header
|
1995-03-09
|
65KB
|
2,427 lines
/* vi:ts=4:sw=4
*
* VIM - Vi IMproved by Bram Moolenaar
*
* Read the file "credits.txt" for a list of people who contributed.
* Read the file "uganda.txt" for copying and usage conditions.
*/
/* for debugging */
/* #define CHECK(c, s) if (c) EMSG(s) */
#define CHECK(c, s)
/*
* memline.c: Contains the functions for appending, deleting and changing the
* text lines. The memfile functions are used to store the information in blocks
* of memory, backed up by a file. The structure of the information is a tree.
* The root of the tree is a pointer block. The leaves of the tree are data
* blocks. In between may be several layers of pointer blocks, forming branches.
*
* Three types of blocks are used:
* - Block nr 0 contains information for recovery
* - Pointer blocks contain list of pointers to other blocks.
* - Data blocks contain the actual text.
*
* Block nr 0 contains the block0 structure (see below).
*
* Block nr 1 is the first pointer block. It is the root of the tree.
* Other pointer blocks are branches.
*
* If a line is too big to fit in a single page, the block
* containing that line is made big enough to hold the line. It may span
* several pages. Otherwise all blocks are one page.
*
* A data block that was filled when starting to edit a file and was not
* changed since then, can have a negative block number. This means that it
* has not yet been assigned a place in the file. When recovering, the lines
* in this data block can be read from the original file. When the block is
* changed (lines appended/deleted/changd) or when it is flushed it gets
* a positive number. Use mf_trans_del() to get the new number, before
* calling mf_get().
*/
/*
* The following functions are available to work with a memline:
*
* ml_open() open a new memline for a buffer
* ml_close() close the memline for a buffer
* ml_close_all() close all memlines
* ml_recover() read a memline for recovery
* ml_sync_all() flush changed blocks to file for all files
* ml_preserve() flush everything into the file for one file
* ml_get() get a pointer to a line
* ml_get_pos() get a pointer to a position in a line
* ml_get_cursor() get a pointer to the char under the cursor
* ml_get_buf() get a pointer to a line in a specific buffer
* ml_line_alloced() return TRUE if line was allocated
* ml_append() append a new line
* ml_replace() replace a line
* ml_delete() delete a line
* ml_setmarked() set mark for a line (for :global command)
* ml_firstmarked() get first line with a mark (for :global command)
* ml_clearmarked() clear all line marks (for :global command)
*/
#include "vim.h"
#include "globals.h"
#include "proto.h"
#include "param.h"
#include <fcntl.h>
#ifdef SASC
# include <proto/dos.h> /* for Open() and Close() */
#endif
typedef struct block0 ZERO_BL; /* contents of the first block */
typedef struct pointer_block PTR_BL; /* contents of a pointer block */
typedef struct data_block DATA_BL; /* contents of a data block */
typedef struct pointer_entry PTR_EN; /* block/line-count pair */
#define DATA_ID (('d' << 8) + 'a') /* data block id */
#define PTR_ID (('p' << 8) + 't') /* pointer block id */
#define BLOCK0_ID (('b' << 8) + '0') /* block 0 id */
/*
* pointer to a block, used in a pointer block
*/
struct pointer_entry
{
blocknr_t pe_bnum; /* block number */
linenr_t pe_line_count; /* number of lines in this branch */
linenr_t pe_old_lnum; /* lnum for this block (for recovery) */
int pe_page_count; /* number of pages in block pe_bnum */
};
/*
* A pointer block contains a list of branches in the tree.
*/
struct pointer_block
{
short_u pb_id; /* id for pointer block: PBL_ID */
short_u pb_count; /* number of pointer in this block */
short_u pb_count_max; /* maximum value for pb_count */
PTR_EN pb_pointer[1]; /* list of pointers to blocks (actually longer)
* followed by empty space until end of page */
};
/*
* A data block is a leaf in the tree.
*
* The text of the lines is at the end of the block. The text of the first line
* in the block is put at the end, the text of the second line in front of it,
* etc. Thus the order of the lines is the opposite of the line number.
*/
struct data_block
{
short_u db_id; /* id for data block: DBL_ID */
unsigned db_free; /* free space available */
unsigned db_txt_start; /* byte where text starts */
unsigned db_txt_end; /* byte just after data block */
linenr_t db_line_count; /* number of lines in this block */
unsigned db_index[1]; /* index for start of line (actually bigger)
* followed by empty space upto db_txt_start
* followed by the text in the lines until
* end of page */
};
/*
* The low bits of db_index hold the actual index. The topmost bit is
* used for the global command to be able to mark a line.
* This method is not clean, but otherwise there would be at least one extra
* byte used for each line.
* The mark has to be in this place to keep it with the correct line when other
* lines are inserted or deleted.
*/
#define DB_MARKED (1 << ((sizeof(unsigned) * 8) - 1))
#define DB_INDEX_MASK (~DB_MARKED)
#define INDEX_SIZE (sizeof(unsigned)) /* size of one db_index entry */
#define HEADER_SIZE (sizeof(DATA_BL) - INDEX_SIZE) /* size of data block header */
struct block0
{
short_u b0_id; /* id for block 0: BLOCK0_ID */
char_u b0_version[10]; /* Vim version string */
int b0_page_size; /* number of bytes per page */
long b0_mtime; /* last modification time of file */
char_u b0_fname[1000]; /* file name of file being edited */
};
/* NOTE: if size of block0 changes, adjust minimal block size in mf_open()!! */
#define STACK_INCR 5 /* number of entries added to ml_stack at a time */
/*
* The line number where the first mark may be is remembered.
* If it is 0 there are no marks at all.
* (always used for the current buffer only, no buffer change possible while
* executing a global command).
*/
static linenr_t lowest_marked = 0;
/*
* arguments for ml_find_line()
*/
#define ML_DELETE 0x11 /* delete line */
#define ML_INSERT 0x12 /* insert line */
#define ML_FIND 0x13 /* just find the line */
#define ML_FLUSH 0x02 /* flush locked block */
#define ML_SIMPLE(x) (x & 0x10) /* DEL, INS or FIND */
static int ml_append_int __ARGS((BUF *, linenr_t, char_u *, colnr_t, int));
static int ml_delete_int __ARGS((BUF *, linenr_t));
static char_u *findswapname __ARGS((BUF *, int));
static void ml_flush_line __ARGS((BUF *));
static BHDR *ml_new_data __ARGS((MEMFILE *, int, int));
static BHDR *ml_new_ptr __ARGS((MEMFILE *));
static BHDR *ml_find_line __ARGS((BUF *, linenr_t, int));
static int ml_add_stack __ARGS((BUF *));
static char_u *makeswapname __ARGS((BUF *, int));
static void ml_lineadd __ARGS((BUF *, int));
/*
* open a new memline for 'curbuf'
*
* return FAIL for failure, OK otherwise
*/
int
ml_open()
{
MEMFILE *mfp = NULL;
char_u *fname = NULL;
BHDR *hp = NULL;
ZERO_BL *b0p;
struct stat st;
int i;
/*
* init fields in memline struct
*/
curbuf->b_ml.ml_stack_size = 0; /* no stack yet */
curbuf->b_ml.ml_stack = NULL; /* no stack yet */
curbuf->b_ml.ml_stack_top = 0; /* nothing in the stack */
curbuf->b_ml.ml_locked = NULL; /* no cached block */
curbuf->b_ml.ml_line_lnum = 0; /* no cached line */
/*
* make fname for swap file
* If we are unable to find a file name, mf_fname will be NULL
* and the memfile will be in memory only (no recovery possible).
* When 'updatecount' is 0 there is never a swap file.
*/
if (p_uc == 0)
fname = NULL;
else
fname = findswapname(curbuf, FALSE); /* NULL detected below */
/*
* open the memfile
*
* If a file name given, 'directory' option is set and does not start with '>'
* may try twice: first in current dir and if that fails in 'directory'.
*/
if (fname != NULL && *p_dir != NUL && *p_dir != '>')
i = 0; /* try twice */
else
i = 1; /* try once */
for ( ; i < 2 && (mfp = mf_open(fname, TRUE, i == 0)) == NULL; ++i)
{
fname = findswapname(curbuf, TRUE); /* NULL detected below */
}
if (mfp == NULL)
goto error;
curbuf->b_ml.ml_mfp = mfp;
curbuf->b_neverloaded = FALSE;
if (p_uc != 0 && mfp->mf_fname == NULL)
{
/* call wait_return if not done by emsg() */
if (EMSG("Unable to open swap file, recovery impossible"))
{
msg_outchar('\n');
wait_return(FALSE);
}
}
/*
* fill block0 struct and write page 0
*/
if ((hp = mf_new(mfp, FALSE, 1)) == NULL)
goto error;
if (hp->bh_bnum != 0)
{
EMSG("didn't get block nr 0?");
goto error;
}
b0p = (ZERO_BL *)(hp->bh_data);
b0p->b0_id = BLOCK0_ID;
STRNCPY(b0p->b0_version, Version, (size_t)10);
if (curbuf->b_filename != NULL)
{
STRNCPY(b0p->b0_fname, curbuf->b_filename, (size_t)1000);
if (stat((char *)curbuf->b_filename, &st) != -1)
b0p->b0_mtime = st.st_mtime;
else
b0p->b0_mtime = 0;
}
else
b0p->b0_fname[0] = NUL;
b0p->b0_page_size = mfp->mf_page_size;
mf_put(mfp, hp, TRUE, FALSE);
/*
* fill in root pointer block and write page 1
*/
if ((hp = ml_new_ptr(mfp)) == NULL)
goto error;
if (hp->bh_bnum != 1)
{
EMSG("didn't get block nr 1?");
goto error;
}
mf_put(mfp, hp, TRUE, FALSE);
curbuf->b_ml.ml_flags = ML_EMPTY;
return OK;
error:
if (mfp != NULL)
{
if (hp)
mf_put(mfp, hp, FALSE, FALSE);
mf_close(mfp, TRUE); /* will also free(fname) */
}
else
free(fname);
curbuf->b_ml.ml_mfp = NULL;
return FAIL;
}
/*
* Open a file for the memfile for all buffers.
* Used when 'updatecount' changes from zero to non-zero.
*/
void
ml_open_files()
{
BUF *buf;
MEMFILE *mfp;
char_u *fname;
int i;
for (buf = firstbuf; buf != NULL; buf = buf->b_next)
{
mfp = buf->b_ml.ml_mfp;
if (mfp == NULL || mfp->mf_fd >= 0) /* nothing to do */
continue;
/*
* make fname for swap file
* If we are unable to find a file name, mf_fname will be NULL
* and the memfile will remain in memory only (no recovery possible).
*/
fname = findswapname(buf, FALSE); /* NULL detected below */
/*
* open the memfile
*
* If a file name given, 'directory' option is set and does not start with '>'
* may try twice: first in current dir and if that fails in 'directory'.
*/
if (fname != NULL && *p_dir != NUL && *p_dir != '>')
i = 0; /* try twice */
else
i = 1; /* try once */
for ( ; i < 2 && mf_open_file(mfp, fname) == FAIL; ++i)
{
fname = findswapname(buf, TRUE); /* NULL detected below */
}
if (mfp->mf_fname == NULL)
{
/* call wait_return if not done by emsg() */
if (EMSG2("Unable to open swap file for \"%s\", recovery impossible",
buf->b_xfilename == NULL ? (char_u *)"No File"
: buf->b_xfilename))
{
msg_outchar('\n');
wait_return(FALSE);
}
}
}
}
/*
* close memline for buffer 'buf' and delete the swap file
*/
void
ml_close(buf)
BUF *buf;
{
if (buf->b_ml.ml_mfp == NULL) /* not open */
return;
mf_close(buf->b_ml.ml_mfp, TRUE); /* delete the .swp file */
if (buf->b_ml.ml_line_lnum != 0 && (buf->b_ml.ml_flags & ML_LINE_DIRTY))
free(buf->b_ml.ml_line_ptr);
free(buf->b_ml.ml_stack);
buf->b_ml.ml_mfp = NULL;
}
/*
* Close all existing memlines and memfiles.
* Used when exiting.
*/
void
ml_close_all()
{
BUF *buf;
for (buf = firstbuf; buf != NULL; buf = buf->b_next)
ml_close(buf);
}
/*
* Update the timestamp in the .swp file.
* Used when the file has been written.
*/
void
ml_timestamp(buf)
BUF *buf;
{
MEMFILE *mfp = NULL;
BHDR *hp = NULL;
ZERO_BL *b0p;
struct stat st;
mfp = buf->b_ml.ml_mfp;
if (mfp == NULL || (hp = mf_get(mfp, (blocknr_t)0, 1)) == NULL)
return;
b0p = (ZERO_BL *)(hp->bh_data);
if (b0p->b0_id != BLOCK0_ID)
{
EMSG("ml_timestamp: Didn't get block 0??");
goto error;
}
/* copy filename again, it may have been changed */
STRNCPY(b0p->b0_fname, buf->b_filename, (size_t)1000);
if (stat((char *)buf->b_filename, &st) != -1)
b0p->b0_mtime = st.st_mtime;
error:
mf_put(mfp, hp, TRUE, FALSE);
}
/*
* try to recover curbuf from the .swp file
*/
void
ml_recover()
{
BUF *buf = NULL;
MEMFILE *mfp = NULL;
char_u *fname = NULL;
BHDR *hp = NULL;
ZERO_BL *b0p;
PTR_BL *pp;
DATA_BL *dp;
IPTR *ip;
blocknr_t bnum;
int page_count;
struct stat org_stat, swp_stat;
int len;
int directly;
linenr_t lnum;
char_u *p;
int i;
long error;
int cannot_open;
linenr_t line_count;
int has_error;
int idx;
int top;
int txt_start;
long size;
/*
* If the file name ends in ".sw?" we use it directly.
* Otherwise ".swp" is appended.
*/
fname = curbuf->b_xfilename;
len = STRLEN(fname);
if (len >= 4 && vim_strnicmp(fname + len - 4, (char_u *)".sw", (size_t)3) == 0)
{
fname = strsave(fname); /* make a copy for mf_open */
directly = TRUE;
}
else
{
fname = makeswapname(curbuf, FALSE);
directly = FALSE;
}
if (fname == NULL)
goto theend; /* out of memory */
/*
* allocate a buffer structure (only the memline in it is really used)
*/
buf = (BUF *)alloc((unsigned)sizeof(BUF));
if (buf == NULL)
goto theend;
/*
* init fields in memline struct
*/
buf->b_ml.ml_stack_size = 0; /* no stack yet */
buf->b_ml.ml_stack = NULL; /* no stack yet */
buf->b_ml.ml_stack_top = 0; /* nothing in the stack */
buf->b_ml.ml_line_lnum = 0; /* no cached line */
buf->b_ml.ml_locked = NULL; /* no locked block */
buf->b_ml.ml_flags = 0;
/*
* open the memfile
*
* If swap file name not given directly, 'directory' option is set and
* does not start with '>' may try twice: first in current dir and if that
* fails in 'directory'.
*/
if (!directly && *p_dir != NUL && *p_dir != '>')
i = 0; /* try twice */
else
i = 1; /* try once */
for ( ; i < 2 && (mfp = mf_open(fname, FALSE, i == 0)) == NULL; i++)
{
fname = makeswapname(curbuf, TRUE);
if (fname == NULL)
goto theend;
}
if (mfp == NULL || mfp->mf_fd < 0)
{
EMSG2("Cannot open %s", fname);
goto theend;
}
buf->b_ml.ml_mfp = mfp;
/*
* try to read block 0
*/
if ((hp = mf_get(mfp, (blocknr_t)0, 1)) == NULL)
{
msg_start();
msg_outstr((char_u *)"Unable to read block 0 from ");
msg_outstr(fname);
msg_outstr((char_u *)"\nMaybe no changes were made or Vim did not update the .swp file");
msg_end();
goto theend;
}
b0p = (ZERO_BL *)(hp->bh_data);
if (b0p->b0_id != BLOCK0_ID)
{
EMSG2("%s is not a swap file", fname);
goto theend;
}
/*
* If we guessed the wrong page size, we have to recalculate the
* hightest block number in the file
*/
if (mfp->mf_page_size != b0p->b0_page_size)
{
mfp->mf_page_size = b0p->b0_page_size;
if ((size = lseek(mfp->mf_fd, 0L, SEEK_END)) <= 0)
mfp->mf_blocknr_max = 0; /* no file or empty file */
else
mfp->mf_blocknr_max = size / mfp->mf_page_size;
mfp->mf_infile_count = mfp->mf_blocknr_max;
}
/*
* If .swp file name given directly, use name from swap file for buffer
*/
if (directly && setfname(b0p->b0_fname, NULL, TRUE) == FAIL)
goto theend;
smsg((char_u *)"Using swap file \"%s\", original file \"%s\"", fname,
curbuf->b_filename == NULL ? "No File" : (char *)curbuf->b_filename);
/*
* check date of swap file and original file
*/
if (curbuf->b_filename != NULL &&
stat((char *)curbuf->b_filename, &org_stat) != -1 &&
((stat((char *)fname, &swp_stat) != -1 &&
org_stat.st_mtime > swp_stat.st_mtime) ||
org_stat.st_mtime != b0p->b0_mtime))
{
MSG("Warning: Original file may have been changed");
}
mf_put(mfp, hp, FALSE, FALSE); /* release block 0 */
hp = NULL;
bnum = 1; /* start with block 1 */
page_count = 1; /* which is 1 page */
lnum = 0; /* append after line 0 in curbuf */
line_count = 0;
idx = 0; /* start with first index in block 1 */
error = 0;
buf->b_ml.ml_stack_top = 0;
buf->b_ml.ml_stack = NULL;
buf->b_ml.ml_stack_size = 0; /* no stack yet */
if (curbuf->b_filename == NULL)
cannot_open = TRUE;
else
cannot_open = FALSE;
for (;;)
{
if ((lnum & 15) == 0) /* check for interrupt now and then */
{
breakcheck();
if (got_int)
break;
}
if (hp != NULL)
mf_put(mfp, hp, FALSE, FALSE); /* release previous block */
/*
* get block
*/
if ((hp = mf_get(mfp, (blocknr_t)bnum, page_count)) == NULL)
{
if (bnum == 1)
{
EMSG2("Unable to read block 1 from %s", fname);
goto theend;
}
++error;
ml_append(lnum++, (char_u *)"???MANY LINES MISSING", (colnr_t)0, TRUE);
}
else /* there is a block */
{
pp = (PTR_BL *)(hp->bh_data);
if (pp->pb_id == PTR_ID) /* it is a pointer block */
{
/* check line count when using pointer block first time */
if (idx == 0 && line_count != 0)
{
for (i = 0; i < (int)pp->pb_count; ++i)
line_count -= pp->pb_pointer[i].pe_line_count;
if (line_count != 0)
{
++error;
ml_append(lnum++, (char_u *)"???LINE COUNT WRONG", (colnr_t)0, TRUE);
}
}
if (pp->pb_count == 0)
{
ml_append(lnum++, (char_u *)"???EMPTY BLOCK", (colnr_t)0, TRUE);
++error;
}
else if (idx < (int)pp->pb_count) /* go a block deeper */
{
if (pp->pb_pointer[idx].pe_bnum < 0)
{
/*
* data block with negative block number
* Try to read lines from the original file.
* This is slow, but it works.
*/
if (!cannot_open)
{
line_count = pp->pb_pointer[idx].pe_line_count;
if (readfile(curbuf->b_filename, NULL, lnum, FALSE,
pp->pb_pointer[idx].pe_old_lnum - 1,
line_count) == FAIL)
cannot_open = TRUE;
else
lnum += line_count;
}
if (cannot_open)
{
++error;
ml_append(lnum++, (char_u *)"???LINES MISSING", (colnr_t)0, TRUE);
}
++idx; /* get same block again for next index */
continue;
}
/*
* going one block deeper in the tree
*/
if ((top = ml_add_stack(buf)) < 0) /* new entry in stack */
{
++error;
break; /* out of memory */
}
ip = &(buf->b_ml.ml_stack[top]);
ip->ip_bnum = bnum;
ip->ip_index = idx;
bnum = pp->pb_pointer[idx].pe_bnum;
line_count = pp->pb_pointer[idx].pe_line_count;
page_count = pp->pb_pointer[idx].pe_page_count;
continue;
}
}
else /* not a pointer block */
{
dp = (DATA_BL *)(hp->bh_data);
if (dp->db_id != DATA_ID) /* block id wrong */
{
if (bnum == 1)
{
EMSG2("Block 1 ID wrong (%s not a .swp file?)", fname);
goto theend;
}
++error;
ml_append(lnum++, (char_u *)"???BLOCK MISSING", (colnr_t)0, TRUE);
}
else
{
/*
* it is a data block
* Append all the lines in this block
*/
has_error = FALSE;
/*
* check length of block
* if wrong, use length in pointer block
*/
if (page_count * mfp->mf_page_size != dp->db_txt_end)
{
ml_append(lnum++, (char_u *)"???until END lines may be messed up", (colnr_t)0, TRUE);
++error;
has_error = TRUE;
dp->db_txt_end = page_count * mfp->mf_page_size;
}
/* make sure there is a NUL at the end of the block */
*((char_u *)dp + dp->db_txt_end - 1) = NUL;
/*
* check number of lines in block
* if wrong, use count in data block
*/
if (line_count != dp->db_line_count)
{
ml_append(lnum++, (char_u *)"???until END lines may have been inserted/deleted", (colnr_t)0, TRUE);
++error;
has_error = TRUE;
}
for (i = 0; i < dp->db_line_count; ++i)
{
txt_start = (dp->db_index[i] & DB_INDEX_MASK);
if (txt_start <= HEADER_SIZE || txt_start >= (int)dp->db_txt_end)
{
p = (char_u *)"???";
++error;
}
else
p = (char_u *)dp + txt_start;
ml_append(lnum++, p, (colnr_t)0, TRUE);
}
if (has_error)
ml_append(lnum++, (char_u *)"???END", (colnr_t)0, TRUE);
}
}
}
if (buf->b_ml.ml_stack_top == 0) /* finished */
break;
/*
* go one block up in the tree
*/
ip = &(buf->b_ml.ml_stack[--(buf->b_ml.ml_stack_top)]);
bnum = ip->ip_bnum;
idx = ip->ip_index + 1; /* go to next index */
page_count = 1;
}
recoverymode = 0;
if (got_int)
EMSG("Recovery Interrupted");
else if (error)
EMSG("Errors detected while recovering; look for lines starting with ???");
else
MSG("Recovery completed; If everything is OK: Save this file and delete the .swp file");
theend:
if (mfp != NULL)
{
if (hp != NULL)
mf_put(mfp, hp, FALSE, FALSE);
mf_close(mfp, FALSE); /* will also free(fname) */
}
else
free(fname);
free(buf);
return;
}
/*
* sync all memlines
*
* Stop syncing when character becomes available, but always sync at
* least one block.
* If 'check_file' is TRUE, check if original file exists and was not changed.
*/
void
ml_sync_all(check_file)
int check_file;
{
BUF *buf;
struct stat st;
for (buf = firstbuf; buf != NULL; buf = buf->b_next)
{
if (buf->b_ml.ml_mfp == NULL || buf->b_ml.ml_mfp->mf_fname == NULL)
continue; /* no file */
ml_flush_line(buf); /* flush buffered line */
(void)ml_find_line(buf, (linenr_t)0, ML_FLUSH); /* flush locked block */
if (buf->b_changed && check_file && mf_need_trans(buf->b_ml.ml_mfp) &&
buf->b_filename != NULL)
{
/*
* if original file does not exist anymore or has been changed
* call ml_preserve to get rid of all negative numbered blocks
*/
if (stat((char *)buf->b_filename, &st) == -1 ||
st.st_mtime != buf->b_mtime)
ml_preserve(buf, FALSE);
}
if (buf->b_ml.ml_mfp->mf_dirty)
{
mf_sync(buf->b_ml.ml_mfp, FALSE, TRUE);
if (mch_char_avail()) /* character available now */
break;
}
}
}
/*
* sync one buffer, including negative blocks
*
* after this all the blocks are in the swap file
*
* Used for the :preserve command and when the original file has been
* changed or deleted.
*
* when message is TRUE the success of preserving is reported
*/
void
ml_preserve(buf, message)
BUF *buf;
int message;
{
BHDR *hp;
linenr_t lnum;
MEMFILE *mfp = buf->b_ml.ml_mfp;
int status;
if (mfp == NULL || mfp->mf_fname == NULL)
{
if (message)
EMSG("Cannot preserve, there is no swap file");
return;
}
ml_flush_line(buf); /* flush buffered line */
(void)ml_find_line(buf, (linenr_t)0, ML_FLUSH); /* flush locked block */
status = mf_sync(mfp, TRUE, FALSE);
/* stack is invalid after mf_sync(.., TRUE, ..) */
buf->b_ml.ml_stack_top = 0;
/*
* Some of the data blocks may have been changed from negative to
* positive block number. In that case the pointer blocks need to be updated.
*
* We don't know in which pointer block the references are, so we visit
* all data blocks until there are no more translations to be done.
* ml_find_line() does the work by translating the negative block numbers
* when getting the first line of each data block.
*/
if (mf_need_trans(mfp))
{
lnum = 1;
while (mf_need_trans(mfp))
{
hp = ml_find_line(buf, lnum, ML_FIND);
if (hp == NULL)
{
status = FAIL;
goto theend;
}
CHECK(buf->b_ml.ml_locked_low != lnum, "low != lnum");
lnum = buf->b_ml.ml_locked_high + 1;
}
(void)ml_find_line(buf, (linenr_t)0, ML_FLUSH); /* flush locked block */
if (mf_sync(mfp, TRUE, FALSE) == FAIL) /* sync the updated pointer blocks */
status = FAIL;
buf->b_ml.ml_stack_top = 0; /* stack is invalid now */
}
theend:
if (message)
{
if (status == OK)
MSG("File preserved");
else
EMSG("Preserve failed");
}
}
/*
* get a pointer to a (read-only copy of a) line
*
* On failure an error message is given and IObuff is returned (to avoid
* having to check for error everywhere).
*/
char_u *
ml_get(lnum)
linenr_t lnum;
{
return ml_get_buf(curbuf, lnum, FALSE);
}
/*
* ml_get_pos: get pointer to position 'pos'
*/
char_u *
ml_get_pos(pos)
FPOS *pos;
{
return (ml_get_buf(curbuf, pos->lnum, FALSE) + pos->col);
}
char_u *
ml_get_cursor()
{
return (ml_get_buf(curbuf, curwin->w_cursor.lnum, FALSE) + curwin->w_cursor.col);
}
/*
* get a pointer to a line in a specific buffer
*
* will_change: if TRUE mark the buffer dirty (chars in the line will be changed)
*/
char_u *
ml_get_buf(buf, lnum, will_change)
BUF *buf;
linenr_t lnum;
int will_change; /* line will be changed */
{
BHDR *hp;
DATA_BL *dp;
char_u *ptr;
if (lnum < 1 || lnum > buf->b_ml.ml_line_count) /* invalid line number */
{
emsg2((char_u *)"ml_get: invalid lnum: %ld", (char_u *)lnum);
errorret:
STRCPY(IObuff, "???");
return IObuff;
}
/*
* See if it is the same line as requested last time.
* Otherwise may need to flush last used line.
*/
if (buf->b_ml.ml_line_lnum != lnum)
{
ml_flush_line(buf);
if (buf->b_ml.ml_flags & ML_EMPTY) /* empty buffer */
return (char_u *)"";
/*
* find the data block containing the line
* This also fills the stack with the blocks from the root to the data block
* This also releases any locked block.
*/
if ((hp = ml_find_line(buf, lnum, ML_FIND)) == NULL)
{
emsg2((char_u *)"ml_get: cannot find line %ld", (char_u *)lnum);
goto errorret;
}
dp = (DATA_BL *)(hp->bh_data);
ptr = (char_u *)dp + ((dp->db_index[lnum - buf->b_ml.ml_locked_low]) & DB_INDEX_MASK);
buf->b_ml.ml_line_ptr = ptr;
buf->b_ml.ml_line_lnum = lnum;
buf->b_ml.ml_flags &= ~ML_LINE_DIRTY;
}
if (will_change)
buf->b_ml.ml_flags |= (ML_LOCKED_DIRTY | ML_LOCKED_POS);
return buf->b_ml.ml_line_ptr;
}
/*
* Check if a line that was just obtained by a call to ml_get
* is in allocated memory.
*/
int
ml_line_alloced()
{
return (curbuf->b_ml.ml_flags & ML_LINE_DIRTY);
}
/*
* append a line after lnum (may be 0 to insert a line in front of the file)
*
* newfile: TRUE when starting to edit a new file, meaning that pe_old_lnum
* will be set for recovery
*
* return FAIL for failure, OK otherwise
*/
int
ml_append(lnum, line, len, newfile)
linenr_t lnum; /* append after this line (can be 0) */
char_u *line; /* text of the new line */
colnr_t len; /* length of new line, including NUL, or 0 */
int newfile; /* flag, see above */
{
if (curbuf->b_ml.ml_line_lnum != 0)
ml_flush_line(curbuf);
return ml_append_int(curbuf, lnum, line, len, newfile);
}
static int
ml_append_int(buf, lnum, line, len, newfile)
BUF *buf;
linenr_t lnum; /* append after this line (can be 0) */
char_u *line; /* text of the new line */
colnr_t len; /* length of line, including NUL, or 0 */
int newfile; /* flag, see above */
{
int i;
int line_count; /* number of indexes in current block */
int offset;
int from, to;
int space_needed; /* space needed for new line */
int page_size;
int page_count;
int db_idx; /* index for lnum in data block */
BHDR *hp = NULL;
BHDR *hp2;
MEMFILE *mfp;
DATA_BL *dp;
PTR_BL *pp;
IPTR *ip;
if (lnum > buf->b_ml.ml_line_count) /* lnum out of range */
return FAIL;
if (lowest_marked && lowest_marked > lnum)
lowest_marked = lnum + 1;
if (len == 0)
len = STRLEN(line) + 1; /* space needed for the text */
space_needed = len + INDEX_SIZE; /* space needed for text + index */
mfp = buf->b_ml.ml_mfp;
page_size = mfp->mf_page_size;
if (buf->b_ml.ml_flags & ML_EMPTY) /* empty file */
{
/*
* Special case: Add first line to empty file.
* Create the first data block.
* If lnum == 0, line 1 is inserted below.
* If lnum == 1, insert an empty line 1 and insert line 2 below.
*/
/*
* allocate the first data block
*/
if (lnum == 1) /* reserve space for line 1 */
space_needed += 1 + INDEX_SIZE;
page_count = ((space_needed + HEADER_SIZE) + page_size - 1) / page_size;
if ((hp = ml_new_data(mfp, newfile, page_count)) == NULL)
return FAIL;
if (lnum == 1) /* insert line 1 here, empty */
{
dp = (DATA_BL *)(hp->bh_data);
dp->db_index[0] = --dp->db_txt_start; /* at end of block */
dp->db_free -= 1 + INDEX_SIZE;
dp->db_line_count = 1;
*((char_u *)dp + dp->db_txt_start) = NUL; /* emtpy line */
space_needed -= 1 + INDEX_SIZE; /* space for line 1 */
}
/*
* update the first pointer block
*/
if ((hp2 = mf_get(mfp, (blocknr_t)1, 1)) == NULL)
{
mf_free(mfp, hp);
return FAIL;
}
pp = (PTR_BL *)(hp2->bh_data); /* must be pointer block */
if (pp->pb_id != PTR_ID)
{
EMSG("pointer block id wrong 5");
mf_free(mfp, hp);
mf_put(mfp, hp2, FALSE, FALSE);
return FAIL;
}
pp->pb_count = 1;
pp->pb_pointer[0].pe_bnum = hp->bh_bnum;
pp->pb_pointer[0].pe_page_count = page_count;
pp->pb_pointer[0].pe_old_lnum = 1;
pp->pb_pointer[0].pe_line_count = lnum + 1; /* line count after insertion */
mf_put(mfp, hp2, TRUE, FALSE);
buf->b_ml.ml_flags &= ~ML_EMPTY;
line_count = lnum; /* 0 or 1 line in block before insertion */
buf->b_ml.ml_line_count = lnum;
db_idx = lnum - 1; /* append new line after -1 or 0 */
}
else /* not empty file */
{
/*
* find the data block containing the previous line
* This also fills the stack with the blocks from the root to the data block
* This also releases any locked block.
*/
if ((hp = ml_find_line(buf, lnum == 0 ? (linenr_t)1 : lnum, ML_INSERT)) == NULL)
return FAIL;
if (lnum == 0) /* got line one instead, correct db_idx */
db_idx = -1; /* careful, it is negative! */
else
db_idx = lnum - buf->b_ml.ml_locked_low;
/* get line count before the insertion */
line_count = buf->b_ml.ml_locked_high - buf->b_ml.ml_locked_low;
}
dp = (DATA_BL *)(hp->bh_data);
/*
* If
* - there is not enough room in the current block
* - appending to the last line in the block
* - not appending to the last line in the file
* insert in front of the next block.
*/
if ((int)dp->db_free < space_needed && db_idx == line_count - 1 &&
lnum < buf->b_ml.ml_line_count)
{
/*
* Now that the line is not going to be inserted in the block that we
* expected, the line count has to be adjusted in the pointer blocks
* by using ml_locked_lineadd.
*/
--(buf->b_ml.ml_locked_lineadd);
--(buf->b_ml.ml_locked_high);
if ((hp = ml_find_line(buf, lnum + 1, ML_INSERT)) == NULL)
return FAIL;
db_idx = -1; /* careful, it is negative! */
/* get line count before the insertion */
line_count = buf->b_ml.ml_locked_high - buf->b_ml.ml_locked_low;
CHECK(buf->b_ml.ml_locked_low != lnum + 1, "locked_low != lnum + 1");
dp = (DATA_BL *)(hp->bh_data);
}
++buf->b_ml.ml_line_count;
if ((int)dp->db_free >= space_needed) /* enough room in data block */
{
/*
* Insert new line in existing data block, or in data block allocated above.
*/
dp->db_txt_start -= len;
dp->db_free -= space_needed;
++(dp->db_line_count);
/*
* move the text of the lines that follow to the front
* adjust the indexes of the lines that follow
*/
if (line_count > db_idx + 1) /* if there are following lines */
{
/*
* Offset is the start of the previous line.
* This will become the character just after the new line.
*/
if (db_idx < 0)
offset = dp->db_txt_end;
else
offset = ((dp->db_index[db_idx]) & DB_INDEX_MASK);
memmove((char *)dp + dp->db_txt_start,
(char *)dp + dp->db_txt_start + len,
(size_t)(offset - (dp->db_txt_start + len)));
for (i = line_count - 1; i > db_idx; --i)
dp->db_index[i + 1] = dp->db_index[i] - len;
dp->db_index[db_idx + 1] = offset - len;
}
else /* add line at the end */
dp->db_index[db_idx + 1] = dp->db_txt_start;
/*
* copy the text into the block
*/
memmove((char *)dp + dp->db_index[db_idx + 1], (char *)line, (size_t)len);
/*
* Mark the block dirty.
*/
buf->b_ml.ml_flags |= ML_LOCKED_DIRTY;
if (!newfile)
buf->b_ml.ml_flags |= ML_LOCKED_POS;
}
else /* not enough space in data block */
{
/*
* If there is not enough room we have to create a new data block and copy some
* lines into it.
* Then we have to insert an entry in the pointer block.
* If this pointer block also is full, we go up another block, and so on, up
* to the root if necessary.
* The line counts in the pointer blocks have already been adjusted by
* ml_find_line().
*/
int line_count_left, line_count_right;
int page_count_left, page_count_right;
BHDR *hp_left;
BHDR *hp_right;
BHDR *hp_new;
int lines_moved;
int data_moved = 0; /* init to shut up gcc */
int total_moved = 0; /* init to shut up gcc */
DATA_BL *dp_right, *dp_left;
int stack_idx;
int in_left;
int lineadd;
blocknr_t bnum_left, bnum_right;
linenr_t lnum_left, lnum_right;
int pb_idx;
PTR_BL *pp_new;
/*
* We are going to allocate a new data block. Depending on the situation
* it will be put to the left or right of the existing block.
* If possible we put the new line in the left block and move the
* lines after it to the right block. Otherwise the new line is also put
* in the right block. This method is more efficient when inserting a lot
* of lines at one place.
*/
if (db_idx < 0) /* left block is new, right block is existing */
{
lines_moved = 0;
in_left = TRUE;
/* space_needed does not change */
}
else /* left block is existing, right block is new */
{
lines_moved = line_count - db_idx - 1;
if (lines_moved == 0)
in_left = FALSE; /* put new line in right block */
/* space_needed does not change */
else
{
data_moved = ((dp->db_index[db_idx]) & DB_INDEX_MASK) - dp->db_txt_start;
total_moved = data_moved + lines_moved * INDEX_SIZE;
if ((int)dp->db_free + total_moved >= space_needed)
{
in_left = TRUE; /* put new line in left block */
space_needed = total_moved;
}
else
{
in_left = FALSE; /* put new line in right block */
space_needed += total_moved;
}
}
}
page_count = ((space_needed + HEADER_SIZE) + page_size - 1) / page_size;
if ((hp_new = ml_new_data(mfp, newfile, page_count)) == 0)
{
/* correct line counts in pointer blocks */
--(buf->b_ml.ml_locked_lineadd);
--(buf->b_ml.ml_locked_high);
return FAIL;
}
if (db_idx < 0) /* left block is new */
{
hp_left = hp_new;
hp_right = hp;
line_count_left = 0;
line_count_right = line_count;
}
else /* right block is new */
{
hp_left = hp;
hp_right = hp_new;
line_count_left = line_count;
line_count_right = 0;
}
dp_right = (DATA_BL *)(hp_right->bh_data);
dp_left = (DATA_BL *)(hp_left->bh_data);
bnum_left = hp_left->bh_bnum;
bnum_right = hp_right->bh_bnum;
page_count_left = hp_left->bh_page_count;
page_count_right = hp_right->bh_page_count;
/*
* May move the new line into the right/new block.
*/
if (!in_left)
{
dp_right->db_txt_start -= len;
dp_right->db_free -= len + INDEX_SIZE;
dp_right->db_index[0] = dp_right->db_txt_start;
memmove((char *)dp_right + dp_right->db_txt_start, (char *)line, (size_t)len);
++line_count_right;
}
/*
* may move lines from the left/old block to the right/new one.
*/
if (lines_moved)
{
/*
*/
dp_right->db_txt_start -= data_moved;
dp_right->db_free -= total_moved;
memmove((char *)dp_right + dp_right->db_txt_start,
(char *)dp_left + dp_left->db_txt_start,
(size_t)data_moved);
offset = dp_right->db_txt_start - dp_left->db_txt_start;
dp_left->db_txt_start += data_moved;
dp_left->db_free += total_moved;
/*
* update indexes in the new block
*/
for (to = line_count_right, from = db_idx + 1; from < line_count_left; ++from, ++to)
dp_right->db_index[to] = dp->db_index[from] + offset;
line_count_right += lines_moved;
line_count_left -= lines_moved;
}
/*
* May move the new line into the left (old or new) block.
*/
if (in_left)
{
dp_left->db_txt_start -= len;
dp_left->db_free -= len + INDEX_SIZE;
dp_left->db_index[line_count_left] = dp_left->db_txt_start;
memmove((char *)dp_left + dp_left->db_txt_start, (char *)line,
(size_t)len);
++line_count_left;
}
if (db_idx < 0) /* left block is new */
{
lnum_left = lnum + 1;
lnum_right = 0;
}
else /* right block is new */
{
lnum_left = 0;
if (in_left)
lnum_right = lnum + 2;
else
lnum_right = lnum + 1;
}
dp_left->db_line_count = line_count_left;
dp_right->db_line_count = line_count_right;
/*
* release the two data blocks
* The new one (hp_new) already has a correct blocknumber.
* The old one (hp, in ml_locked) gets a positive blocknumber if
* we changed it and we are not editing a new file.
*/
if (lines_moved || in_left)
buf->b_ml.ml_flags |= ML_LOCKED_DIRTY;
if (!newfile && db_idx >= 0 && in_left)
buf->b_ml.ml_flags |= ML_LOCKED_POS;
mf_put(mfp, hp_new, TRUE, FALSE);
/*
* flush the old data block
* set ml_locked_lineadd to 0, because the updating of the
* pointer blocks is done below
*/
lineadd = buf->b_ml.ml_locked_lineadd;
buf->b_ml.ml_locked_lineadd = 0;
ml_find_line(buf, (linenr_t)0, ML_FLUSH); /* flush data block */
/*
* update pointer blocks for the new data block
*/
for (stack_idx = buf->b_ml.ml_stack_top - 1; stack_idx >= 0; --stack_idx)
{
ip = &(buf->b_ml.ml_stack[stack_idx]);
pb_idx = ip->ip_index;
if ((hp = mf_get(mfp, ip->ip_bnum, 1)) == NULL)
return FAIL;
pp = (PTR_BL *)(hp->bh_data); /* must be pointer block */
if (pp->pb_id != PTR_ID)
{
EMSG("pointer block id wrong 3");
mf_put(mfp, hp, FALSE, FALSE);
return FAIL;
}
/*
* TODO: If the pointer block is full and we are adding at the end
* try to insert in front of the next block
*/
if (pp->pb_count < pp->pb_count_max) /* block not full, add one entry */
{
if (pb_idx + 1 < (int)pp->pb_count)
memmove((char *)&pp->pb_pointer[pb_idx + 2],
(char *)&pp->pb_pointer[pb_idx + 1],
(size_t)(pp->pb_count - pb_idx - 1) * sizeof(PTR_EN));
++pp->pb_count;
pp->pb_pointer[pb_idx].pe_line_count = line_count_left;
pp->pb_pointer[pb_idx].pe_bnum = bnum_left;
pp->pb_pointer[pb_idx].pe_page_count = page_count_left;
pp->pb_pointer[pb_idx + 1].pe_line_count = line_count_right;
pp->pb_pointer[pb_idx + 1].pe_bnum = bnum_right;
pp->pb_pointer[pb_idx + 1].pe_page_count = page_count_right;
if (lnum_left != 0)
pp->pb_pointer[pb_idx].pe_old_lnum = lnum_left;
if (lnum_right != 0)
pp->pb_pointer[pb_idx + 1].pe_old_lnum = lnum_right;
mf_put(mfp, hp, TRUE, FALSE);
buf->b_ml.ml_stack_top = stack_idx + 1; /* truncate stack */
if (lineadd)
{
--(buf->b_ml.ml_stack_top);
/* fix line count for rest of blocks in the stack */
ml_lineadd(buf, lineadd);
/* fix stack itself */
buf->b_ml.ml_stack[buf->b_ml.ml_stack_top].ip_high += lineadd;
++(buf->b_ml.ml_stack_top);
}
return OK;
}
else /* pointer block full */
{
/*
* split the pointer block
* allocate a new pointer block
* move some of the pointer into the new block
* prepare for updating the parent block
*/
for (;;) /* do this twice when splitting block 1 */
{
hp_new = ml_new_ptr(mfp);
if (hp_new == NULL) /* TODO: try to fix tree */
return FAIL;
pp_new = (PTR_BL *)(hp_new->bh_data);
if (hp->bh_bnum != 1)
break;
/*
* if block 1 becomes full the tree is given an extra level
* The pointers from block 1 are moved into the new block.
* block 1 is updated to point to the new block
* then continue to split the new block
*/
memmove((char *)pp_new, (char *)pp, (size_t)page_size);
pp->pb_count = 1;
pp->pb_pointer[0].pe_bnum = hp_new->bh_bnum;
pp->pb_pointer[0].pe_line_count = buf->b_ml.ml_line_count;
pp->pb_pointer[0].pe_old_lnum = 1;
pp->pb_pointer[0].pe_page_count = 1;
mf_put(mfp, hp, TRUE, FALSE); /* release block 1 */
hp = hp_new; /* new block is to be split */
pp = pp_new;
CHECK(stack_idx != 0, "stack_idx should be 0");
ip->ip_index = 0;
++stack_idx; /* do block 1 again later */
}
/*
* move the pointers after the current one to the new block
* If there are none, the new entry will be in the new block.
*/
total_moved = pp->pb_count - pb_idx - 1;
if (total_moved)
{
memmove((char *)&pp_new->pb_pointer[0],
(char *)&pp->pb_pointer[pb_idx + 1],
(size_t)(total_moved) * sizeof(PTR_EN));
pp_new->pb_count = total_moved;
pp->pb_count -= total_moved - 1;
pp->pb_pointer[pb_idx + 1].pe_bnum = bnum_right;
pp->pb_pointer[pb_idx + 1].pe_line_count = line_count_right;
pp->pb_pointer[pb_idx + 1].pe_page_count = page_count_right;
if (lnum_right)
pp->pb_pointer[pb_idx + 1].pe_old_lnum = lnum_right;
}
else
{
pp_new->pb_count = 1;
pp_new->pb_pointer[0].pe_bnum = bnum_right;
pp_new->pb_pointer[0].pe_line_count = line_count_right;
pp_new->pb_pointer[0].pe_page_count = page_count_right;
pp_new->pb_pointer[0].pe_old_lnum = lnum_right;
}
pp->pb_pointer[pb_idx].pe_bnum = bnum_left;
pp->pb_pointer[pb_idx].pe_line_count = line_count_left;
pp->pb_pointer[pb_idx].pe_page_count = page_count_left;
if (lnum_left)
pp->pb_pointer[pb_idx].pe_old_lnum = lnum_left;
lnum_left = 0;
lnum_right = 0;
/*
* recompute line counts
*/
line_count_right = 0;
for (i = 0; i < (int)pp_new->pb_count; ++i)
line_count_right += pp_new->pb_pointer[i].pe_line_count;
line_count_left = 0;
for (i = 0; i < (int)pp->pb_count; ++i)
line_count_left += pp->pb_pointer[i].pe_line_count;
bnum_left = hp->bh_bnum;
bnum_right = hp_new->bh_bnum;
page_count_left = 1;
page_count_right = 1;
mf_put(mfp, hp, TRUE, FALSE);
mf_put(mfp, hp_new, TRUE, FALSE);
}
}
EMSG("Updated too many blocks?");
buf->b_ml.ml_stack_top = 0; /* invalidate stack */
}
return OK;
}
/*
* replace line lnum, with buffering, in current buffer
*
* If copy is TRUE, make a copy of the line, otherwise the line has been
* copied to allocated memory already.
*
* return FAIL for failure, OK otherwise
*/
int
ml_replace(lnum, line, copy)
linenr_t lnum;
char_u *line;
int copy;
{
int status;
if (line == NULL) /* just checking... */
return FAIL;
/*
* if empty file simply append the one and only line
*/
if (curbuf->b_ml.ml_flags & ML_EMPTY)
{
if (*line == NUL) /* nothing to do */
return OK;
status = ml_append_int(curbuf, lnum - 1, line, (colnr_t)0, FALSE);
if (!copy)
free(line);
return status;
}
/*
* if only line replaced by empty line, buffer becomes empty
*/
if (lnum == 1 && *line == NUL && curbuf->b_ml.ml_line_count == 1)
{
status = ml_delete(lnum);
if (!copy)
free(line);
return status;
}
if (curbuf->b_ml.ml_line_lnum != lnum) /* other line currently buffered */
ml_flush_line(curbuf); /* flush it */
else if (curbuf->b_ml.ml_flags & ML_LINE_DIRTY) /* same line has been allocated */
free(curbuf->b_ml.ml_line_ptr); /* free it */
if (copy && (line = strsave(line)) == NULL) /* allocate memory for the line */
return FAIL;
curbuf->b_ml.ml_line_ptr = line;
curbuf->b_ml.ml_line_lnum = lnum;
curbuf->b_ml.ml_flags |= ML_LINE_DIRTY;
return OK;
}
/*
* delete line 'lnum'
*
* return FAIL for failure, OK otherwise
*/
int
ml_delete(lnum)
linenr_t lnum;
{
ml_flush_line(curbuf);
return ml_delete_int(curbuf, lnum);
}
static int
ml_delete_int(buf, lnum)
BUF *buf;
linenr_t lnum;
{
BHDR *hp;
MEMFILE *mfp;
DATA_BL *dp;
PTR_BL *pp;
IPTR *ip;
int count; /* number of entries in block */
int idx;
int stack_idx;
int text_start;
int line_start;
int line_size;
int i;
if (lnum < 1 || lnum > buf->b_ml.ml_line_count)
return FAIL;
if (lowest_marked && lowest_marked > lnum)
lowest_marked--;
if (buf->b_ml.ml_flags & ML_EMPTY) /* nothing to delete */
return FAIL;
/*
* find the data block containing the line
* This also fills the stack with the blocks from the root to the data block
* This also releases any locked block.
*/
mfp = buf->b_ml.ml_mfp;
if ((hp = ml_find_line(buf, lnum, ML_DELETE)) == NULL)
return FAIL;
dp = (DATA_BL *)(hp->bh_data);
/* compute line count before the delete */
count = (long)(buf->b_ml.ml_locked_high) - (long)(buf->b_ml.ml_locked_low) + 2;
idx = lnum - buf->b_ml.ml_locked_low;
/*
* If the file becomes empty the data block is freed.
* The pointer blocks will be updated when the first line is appended.
*/
if (--buf->b_ml.ml_line_count == 0) /* file becomes empty */
{
CHECK(count != 1, "ml_delete: count not 1");
CHECK(idx != 0, "ml_delete: idx not 0");
buf->b_ml.ml_line_count = 1;
buf->b_ml.ml_flags |= ML_EMPTY;
mf_free(mfp, hp); /* free the data block */
buf->b_ml.ml_locked = NULL;
buf->b_ml.ml_stack_top = 0; /* flush stack */
return OK;
}
/*
* special case: If there is only one line in the data block it becomes empty.
* Then we have to remove the entry, pointing to this data block, from the
* pointer block. If this pointer block also becomes empty, we go up another
* block, and so on, up to the root if necessary.
* The line counts in the pointer blocks have already been adjusted by
* ml_find_line().
*/
if (count == 1)
{
mf_free(mfp, hp); /* free the data block */
buf->b_ml.ml_locked = NULL;
for (stack_idx = buf->b_ml.ml_stack_top - 1; stack_idx >= 0; --stack_idx)
{
buf->b_ml.ml_stack_top = 0; /* stack is invalid when failing */
ip = &(buf->b_ml.ml_stack[stack_idx]);
idx = ip->ip_index;
if ((hp = mf_get(mfp, ip->ip_bnum, 1)) == NULL)
return FAIL;
pp = (PTR_BL *)(hp->bh_data); /* must be pointer block */
if (pp->pb_id != PTR_ID)
{
EMSG("pointer block id wrong 4");
mf_put(mfp, hp, FALSE, FALSE);
return FAIL;
}
count = --(pp->pb_count);
if (count == 0) /* the pointer block becomes empty! */
mf_free(mfp, hp);
else
{
if (count != idx) /* move entries after the deleted one */
memmove((char *)&pp->pb_pointer[idx],
(char *)&pp->pb_pointer[idx + 1],
(size_t)(count - idx) * sizeof(PTR_EN));
mf_put(mfp, hp, TRUE, FALSE);
buf->b_ml.ml_stack_top = stack_idx; /* truncate stack */
/* fix line count for rest of blocks in the stack */
if (buf->b_ml.ml_locked_lineadd)
{
ml_lineadd(buf, buf->b_ml.ml_locked_lineadd);
buf->b_ml.ml_stack[buf->b_ml.ml_stack_top].ip_high +=
buf->b_ml.ml_locked_lineadd;
}
++(buf->b_ml.ml_stack_top);
return OK;
}
}
CHECK(1, "deleted block 1?");
return OK;
}
/*
* delete the text by moving the next lines forwards
*/
text_start = dp->db_txt_start;
line_start = ((dp->db_index[idx]) & DB_INDEX_MASK);
if (idx == 0) /* first line in block, text at the end */
line_size = dp->db_txt_end - line_start;
else
line_size = ((dp->db_index[idx - 1]) & DB_INDEX_MASK) - line_start;
memmove((char *)dp + text_start + line_size, (char *)dp + text_start,
(size_t)(line_start - text_start));
/*
* delete the index by moving the next indexes backwards
* Adjust the indexes for the text movement.
*/
for (i = idx; i < count - 1; ++i)
dp->db_index[i] = dp->db_index[i + 1] + line_size;
dp->db_free += line_size + INDEX_SIZE;
dp->db_txt_start += line_size;
--(dp->db_line_count);
/*
* mark the block dirty and make sure it is in the file (for recovery)
*/
buf->b_ml.ml_flags |= (ML_LOCKED_DIRTY | ML_LOCKED_POS);
return OK;
}
/*
* set the B_MARKED flag for line 'lnum'
*/
void
ml_setmarked(lnum)
linenr_t lnum;
{
BHDR *hp;
DATA_BL *dp;
if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count) /* invalid line number */
return; /* give error message? */
if (lowest_marked == 0 || lowest_marked > lnum)
lowest_marked = lnum;
/*
* find the data block containing the line
* This also fills the stack with the blocks from the root to the data block
* This also releases any locked block.
*/
if ((hp = ml_find_line(curbuf, lnum, ML_FIND)) == NULL)
return; /* give error message? */
dp = (DATA_BL *)(hp->bh_data);
dp->db_index[lnum - curbuf->b_ml.ml_locked_low] |= DB_MARKED;
curbuf->b_ml.ml_flags |= ML_LOCKED_DIRTY;
}
/*
* find the first line with its B_MARKED flag set
*/
linenr_t
ml_firstmarked()
{
BHDR *hp;
DATA_BL *dp;
linenr_t lnum;
int i;
/*
* The search starts with lowest_marked line. This is the last line where
* a mark was found, adjusted by inserting/deleting lines.
*/
for (lnum = lowest_marked; lnum <= curbuf->b_ml.ml_line_count; )
{
/*
* Find the data block containing the line.
* This also fills the stack with the blocks from the root to the data block
* This also releases any locked block.
*/
if ((hp = ml_find_line(curbuf, lnum, ML_FIND)) == NULL)
return (linenr_t)0; /* give error message? */
dp = (DATA_BL *)(hp->bh_data);
for (i = lnum - curbuf->b_ml.ml_locked_low;
lnum <= curbuf->b_ml.ml_locked_high; ++i, ++lnum)
if ((dp->db_index[i]) & DB_MARKED)
{
(dp->db_index[i]) &= DB_INDEX_MASK;
curbuf->b_ml.ml_flags |= ML_LOCKED_DIRTY;
lowest_marked = lnum + 1;
return lnum;
}
}
return (linenr_t) 0;
}
/*
* return TRUE if line 'lnum' has a mark
*/
int
ml_has_mark(lnum)
linenr_t lnum;
{
BHDR *hp;
DATA_BL *dp;
if ((hp = ml_find_line(curbuf, lnum, ML_FIND)) == NULL)
return FALSE;
dp = (DATA_BL *)(hp->bh_data);
return (int)((dp->db_index[lnum - curbuf->b_ml.ml_locked_low]) & DB_MARKED);
}
/*
* clear all DB_MARKED flags
*/
void
ml_clearmarked()
{
BHDR *hp;
DATA_BL *dp;
linenr_t lnum;
int i;
/*
* The search starts with line lowest_marked.
*/
for (lnum = lowest_marked; lnum <= curbuf->b_ml.ml_line_count; )
{
/*
* Find the data block containing the line.
* This also fills the stack with the blocks from the root to the data block
* This also releases any locked block.
*/
if ((hp = ml_find_line(curbuf, lnum, ML_FIND)) == NULL)
return; /* give error message? */
dp = (DATA_BL *)(hp->bh_data);
for (i = lnum - curbuf->b_ml.ml_locked_low;
lnum <= curbuf->b_ml.ml_locked_high; ++i, ++lnum)
if ((dp->db_index[i]) & DB_MARKED)
{
(dp->db_index[i]) &= DB_INDEX_MASK;
curbuf->b_ml.ml_flags |= ML_LOCKED_DIRTY;
}
}
lowest_marked = 0;
return;
}
/*
* flush ml_line if necessary
*/
static void
ml_flush_line(buf)
BUF *buf;
{
BHDR *hp;
DATA_BL *dp;
linenr_t lnum;
char_u *new_line;
char_u *old_line;
colnr_t new_len;
int old_len;
int extra;
int idx;
int start;
int count;
int i;
if (buf->b_ml.ml_line_lnum == 0) /* nothing to do */
return;
if (buf->b_ml.ml_flags & ML_LINE_DIRTY)
{
lnum = buf->b_ml.ml_line_lnum;
new_line = buf->b_ml.ml_line_ptr;
hp = ml_find_line(buf, lnum, ML_FIND);
if (hp == NULL)
EMSG2("Cannot fine line %ld", (char_u *)lnum);
else
{
dp = (DATA_BL *)(hp->bh_data);
idx = lnum - buf->b_ml.ml_locked_low;
start = ((dp->db_index[idx]) & DB_INDEX_MASK);
old_line = (char_u *)dp + start;
if (idx == 0) /* line is last in block */
old_len = dp->db_txt_end - start;
else /* text of previous line follows */
old_len = (dp->db_index[idx - 1] & DB_INDEX_MASK) - start;
new_len = STRLEN(new_line) + 1;
extra = new_len - old_len; /* negative if lines gets smaller */
/*
* if new line fits in data block, replace directly
*/
if ((int)dp->db_free >= extra)
{
/* if the length changes and there are following lines */
count = buf->b_ml.ml_locked_high - buf->b_ml.ml_locked_low + 1;
if (extra != 0 && idx < count - 1)
{
/* move text of following lines */
memmove((char *)dp + dp->db_txt_start - extra,
(char *)dp + dp->db_txt_start,
(size_t)(start - dp->db_txt_start));
/* adjust pointers of this and following lines */
for (i = idx + 1; i < count; ++i)
dp->db_index[i] -= extra;
}
dp->db_index[idx] -= extra;
/* adjust free space */
dp->db_free -= extra;
dp->db_txt_start -= extra;
/* copy new line into the data block */
memmove((char *)old_line - extra, (char *)new_line, (size_t)new_len);
buf->b_ml.ml_flags |= (ML_LOCKED_DIRTY | ML_LOCKED_POS);
}
else
{
/*
* Cannot do it in one data block: delete and append.
*/
/* How about handling errors??? */
(void)ml_delete_int(buf, lnum);
(void)ml_append_int(buf, lnum - 1, new_line, new_len, FALSE);
}
}
free(new_line);
}
buf->b_ml.ml_line_lnum = 0;
}
/*
* create a new, empty, data block
*/
static BHDR *
ml_new_data(mfp, negative, page_count)
MEMFILE *mfp;
int negative;
int page_count;
{
BHDR *hp;
DATA_BL *dp;
if ((hp = mf_new(mfp, negative, page_count)) == NULL)
return NULL;
dp = (DATA_BL *)(hp->bh_data);
dp->db_id = DATA_ID;
dp->db_txt_start = dp->db_txt_end = page_count * mfp->mf_page_size;
dp->db_free = dp->db_txt_start - HEADER_SIZE;
dp->db_line_count = 0;
return hp;
}
/*
* create a new, empty, pointer block
*/
static BHDR *
ml_new_ptr(mfp)
MEMFILE *mfp;
{
BHDR *hp;
PTR_BL *pp;
if ((hp = mf_new(mfp, FALSE, 1)) == NULL)
return NULL;
pp = (PTR_BL *)(hp->bh_data);
pp->pb_id = PTR_ID;
pp->pb_count = 0;
pp->pb_count_max = (mfp->mf_page_size - sizeof(PTR_BL)) / sizeof(PTR_EN) + 1;
return hp;
}
/*
* lookup line 'lnum' in a memline
*
* action: if ML_DELETE or ML_INSERT the line count is updated while searching
* if ML_FLUSH only flush a locked block
* if ML_FIND just find the line
*
* If the block was found it is locked and put in ml_locked.
* The stack is updated to lead to the locked block. The ip_high field in
* the stack is updated to reflect the last line in the block AFTER the
* insert or delete, also if the pointer block has not been updated yet. But
* if if ml_locked != NULL ml_locked_lineadd must be added to ip_high.
*
* return: NULL for failure, pointer to block header otherwise
*/
static BHDR *
ml_find_line(buf, lnum, action)
BUF *buf;
linenr_t lnum;
int action;
{
DATA_BL *dp;
PTR_BL *pp;
IPTR *ip;
BHDR *hp;
MEMFILE *mfp;
linenr_t t;
blocknr_t bnum, bnum2;
int dirty;
linenr_t low, high;
int top;
int page_count;
int idx;
mfp = buf->b_ml.ml_mfp;
/*
* If there is a locked block check if the wanted line is in it.
* If not, flush and release the locked block.
* Don't do this for ML_INSERT_SAME, because the stack need to be updated.
* Don't do this for ML_FLUSH, because we want to flush the locked block.
*/
if (buf->b_ml.ml_locked)
{
if (ML_SIMPLE(action) && buf->b_ml.ml_locked_low <= lnum &&
buf->b_ml.ml_locked_high >= lnum)
{
/* remember to update pointer blocks and stack later */
if (action == ML_INSERT)
{
++(buf->b_ml.ml_locked_lineadd);
++(buf->b_ml.ml_locked_high);
}
else if (action == ML_DELETE)
{
--(buf->b_ml.ml_locked_lineadd);
--(buf->b_ml.ml_locked_high);
}
return (buf->b_ml.ml_locked);
}
mf_put(mfp, buf->b_ml.ml_locked, buf->b_ml.ml_flags & ML_LOCKED_DIRTY,
buf->b_ml.ml_flags & ML_LOCKED_POS);
buf->b_ml.ml_locked = NULL;
/*
* if lines have been added or deleted in the locked block, need to
* update the line count in pointer blocks
*/
if (buf->b_ml.ml_locked_lineadd)
ml_lineadd(buf, buf->b_ml.ml_locked_lineadd);
}
if (action == ML_FLUSH) /* nothing else to do */
return NULL;
bnum = 1; /* start at the root of the tree */
page_count = 1;
low = 1;
high = buf->b_ml.ml_line_count;
if (action == ML_FIND) /* first try stack entries */
{
for (top = buf->b_ml.ml_stack_top - 1; top >= 0; --top)
{
ip = &(buf->b_ml.ml_stack[top]);
if (ip->ip_low <= lnum && ip->ip_high >= lnum)
{
bnum = ip->ip_bnum;
low = ip->ip_low;
high = ip->ip_high;
buf->b_ml.ml_stack_top = top; /* truncate stack at prev entry */
break;
}
}
if (top < 0)
buf->b_ml.ml_stack_top = 0; /* not found, start at the root */
}
else /* ML_DELETE or ML_INSERT */
buf->b_ml.ml_stack_top = 0; /* start at the root */
/*
* search downwards in the tree until a data block is found
*/
for (;;)
{
if ((hp = mf_get(mfp, bnum, page_count)) == NULL)
goto error_noblock;
/*
* update high for insert/delete
*/
if (action == ML_INSERT)
++high;
else if (action == ML_DELETE)
--high;
dp = (DATA_BL *)(hp->bh_data);
if (dp->db_id == DATA_ID) /* data block */
{
buf->b_ml.ml_locked = hp;
buf->b_ml.ml_locked_low = low;
buf->b_ml.ml_locked_high = high;
buf->b_ml.ml_locked_lineadd = 0;
buf->b_ml.ml_flags &= ~(ML_LOCKED_DIRTY | ML_LOCKED_POS);
return hp;
}
pp = (PTR_BL *)(dp); /* must be pointer block */
if (pp->pb_id != PTR_ID)
{
EMSG("pointer block id wrong");
goto error_block;
}
if ((top = ml_add_stack(buf)) < 0) /* add new entry to stack */
goto error_block;
ip = &(buf->b_ml.ml_stack[top]);
ip->ip_bnum = bnum;
ip->ip_low = low;
ip->ip_high = high;
ip->ip_index = -1; /* index not known yet */
dirty = FALSE;
for (idx = 0; idx < (int)pp->pb_count; ++idx)
{
t = pp->pb_pointer[idx].pe_line_count;
CHECK(t == 0, "pe_line_count is zero");
if ((low += t) > lnum)
{
ip->ip_index = idx;
bnum = pp->pb_pointer[idx].pe_bnum;
page_count = pp->pb_pointer[idx].pe_page_count;
high = low - 1;
low -= t;
/*
* a negative block number may have been changed
*/
if (bnum < 0)
{
bnum2 = mf_trans_del(mfp, bnum);
if (bnum != bnum2)
{
bnum = bnum2;
pp->pb_pointer[idx].pe_bnum = bnum;
dirty = TRUE;
}
}
break;
}
}
if (idx >= (int)pp->pb_count) /* past the end: something wrong! */
{
if (lnum > buf->b_ml.ml_line_count)
emsg2((char_u *)"line number out of range: %ld past the end", (char_u *)(lnum - buf->b_ml.ml_line_count));
else
emsg2((char_u *)"line count wrong in block %ld", (char_u *)(bnum));
goto error_block;
}
if (action == ML_DELETE)
{
pp->pb_pointer[idx].pe_line_count--;
dirty = TRUE;
}
else if (action == ML_INSERT)
{
pp->pb_pointer[idx].pe_line_count++;
dirty = TRUE;
}
mf_put(mfp, hp, dirty, FALSE);
}
error_block:
mf_put(mfp, hp, FALSE, FALSE);
error_noblock:
/*
* If action is ML_DELETE or ML_INSERT we have to correct the tree for
* the incremented/decremented line counts, because there won't be a line
* inserted/deleted after all.
*/
if (action == ML_DELETE)
ml_lineadd(buf, 1);
else if (action == ML_INSERT)
ml_lineadd(buf, -1);
buf->b_ml.ml_stack_top = 0;
return NULL;
}
/*
* add an entry to the info pointer stack
*
* return -1 for failure, number of the new entry otherwise
*/
static int
ml_add_stack(buf)
BUF *buf;
{
int top;
IPTR *newstack;
top = buf->b_ml.ml_stack_top;
/* may have to increase the stack size */
if (top == buf->b_ml.ml_stack_size)
{
CHECK(top > 0, "Stack size increases"); /* more than 5 levels??? */
newstack = (IPTR *)alloc((unsigned)sizeof(IPTR) * (buf->b_ml.ml_stack_size + STACK_INCR));
if (newstack == NULL)
return -1;
memmove((char *)newstack, (char *)buf->b_ml.ml_stack, (size_t)top * sizeof(IPTR));
free(buf->b_ml.ml_stack);
buf->b_ml.ml_stack = newstack;
buf->b_ml.ml_stack_size += STACK_INCR;
}
buf->b_ml.ml_stack_top++;
return top;
}
/*
* Update the pointer blocks on the stack for inserted/deleted lines.
* The stack itself is also updated.
*
* When a insert/delete line action fails, the line is not inserted/deleted,
* but the pointer blocks have already been updated. That is fixed here by
* walking through the stack.
*
* Count is the number of lines added, negative if lines have been deleted.
*/
static void
ml_lineadd(buf, count)
BUF *buf;
int count;
{
int idx;
IPTR *ip;
PTR_BL *pp;
MEMFILE *mfp = buf->b_ml.ml_mfp;
BHDR *hp;
for (idx = buf->b_ml.ml_stack_top - 1; idx >= 0; --idx)
{
ip = &(buf->b_ml.ml_stack[idx]);
if ((hp = mf_get(mfp, ip->ip_bnum, 1)) == NULL)
break;
pp = (PTR_BL *)(hp->bh_data); /* must be pointer block */
if (pp->pb_id != PTR_ID)
{
mf_put(mfp, hp, FALSE, FALSE);
EMSG("pointer block id wrong 2");
break;
}
pp->pb_pointer[ip->ip_index].pe_line_count += count;
ip->ip_high += count;
mf_put(mfp, hp, TRUE, FALSE);
}
}
/*
* make swap file name out of the filename
*/
static char_u *
makeswapname(buf, second_try)
BUF *buf;
int second_try;
{
char_u *r, *s, *fname;
char_u *pdir;
r = modname(curbuf->b_xfilename, (char_u *)".swp");
/*
* do not try to use 'directory' option
* - if 'directory' option not set
* - if out of memory
* - not a second try and 'directory' option does not start with '>'
*/
if (*p_dir == NUL || r == NULL || (!second_try && *p_dir != '>'))
return r;
fname = gettail(r);
if (*p_dir == '>') /* skip '>' in front of dir name */
pdir = p_dir + 1;
else
pdir = p_dir;
s = alloc((unsigned)(STRLEN(pdir) + STRLEN(fname) + 2));
if (s != NULL)
{
STRCPY(s, pdir);
if (*s && !ispathsep(*(s + STRLEN(s) - 1))) /* don't add '/' after ':' */
STRCAT(s, PATHSEPSTR);
STRCAT(s, fname);
}
free(r);
return s;
}
/*
* Find out what name to use for the swap file for buffer 'buf'.
*
* Several names are tried to find one that does not exist
*/
static char_u *
findswapname(buf, second_try)
BUF *buf;
int second_try;
{
char_u *fname;
int n;
#ifdef AMIGA
int r;
BPTR fh;
FILE *dummyfd = NULL;
/*
* If we start editing a new file, e.g. "test.doc", which resides on an MSDOS
* compatible filesystem, it is possible that the file "test.doc.swp" which we
* create will be exactly the same file. To avoid this problem we temporarily
* create "test.doc".
*/
if (!(buf->b_p_sn || buf->b_shortname) && buf->b_xfilename &&
getperm(buf->b_xfilename) < 0)
dummyfd = fopen((char *)buf->b_xfilename, "w");
#endif
/*
* we try different names until we find one that does not exist yet
*/
fname = makeswapname(buf, second_try);
for (;;)
{
if (fname == NULL) /* must be out of memory */
break;
if ((n = STRLEN(fname)) == 0) /* safety check */
{
free(fname);
fname = NULL;
break;
}
/*
* check if the scriptfile already exists
*/
if (getperm(fname) < 0) /* it does not exist */
{
#ifdef AMIGA
fh = Open((UBYTE *)fname, (long)MODE_NEWFILE);
/*
* on the Amiga getperm() will return -1 when the file exists but
* is being used by another program. This happens if you edit
* a file twice.
*/
if (fh != (BPTR)NULL) /* can open file, OK */
{
Close(fh);
break;
}
if (IoErr() != ERROR_OBJECT_IN_USE && IoErr() != ERROR_OBJECT_EXISTS)
#endif
break;
}
/*
* get here when file already exists
*/
if (fname[n - 1] == 'p') /* first try */
{
#ifdef AMIGA
/*
* on MS-DOS compatible filesystems (e.g. messydos) file.doc.swp
* and file.doc are the same file. To guess if this problem is
* present try if file.doc.swx exists. If it does, we set
* buf->b_shortname and try file_doc.swp (dots replaced by
* underscores for this file), and try again. If it doesn't we
* assume that "file.doc.swp" already exists.
*/
if (!(buf->b_p_sn || buf->b_shortname)) /* not tried yet */
{
fname[n - 1] = 'x';
r = getperm(fname); /* try "file.swx" */
fname[n - 1] = 'p';
if (r >= 0) /* it seems to exist */
{
buf->b_shortname = TRUE;
free(fname);
fname = makeswapname(buf, second_try); /* '.' replaced by '_' */
continue; /* try again */
}
}
#endif
/*
* If we get here ".swp" file really exists.
* Give an error message, unless recovering or no file name.
*/
if (!recoverymode && buf->b_xfilename != NULL)
{
/* call wait_return if not done by emsg() */
if (EMSG2(".swp file exists: An edit of file \"%s\" may not have been finished", buf->b_xfilename))
{
msg_outchar('\n');
wait_return(FALSE); /* do call wait_return now */
}
}
}
if (fname[n - 1] == 'a') /* tried enough names, give up */
{
free(fname);
fname = NULL;
break;
}
--fname[n - 1]; /* change last char of the name */
}
#ifdef AMIGA
if (dummyfd) /* file has been created temporarily */
{
fclose(dummyfd);
remove((char *)buf->b_xfilename);
}
#endif
return fname;
}